import taichi as ti

ti.init(debug=True)

tensor = ti.Vector.field(2,dtype=ti.f32,shape=(4,8,16,32,64))

@ti.kernel
def print_tensor_size(x:ti.template()):
    print(x.shape)

print_tensor_size(tensor)